package 剑指Offer.简单;

import Entity.TreeNode;

import java.util.ArrayList;
import java.util.LinkedList;
import java.util.List;
import java.util.Queue;

/**
 * 从上到下按层打印二叉树，同一层的节点按从左到右的顺序打印，每一层打印到一行。
 *
 *  
 *
 * 例如:
 * 给定二叉树: [3,9,20,null,null,15,7],
 *
 *     3
 *    / \
 *   9  20
 *     /  \
 *    15   7
 * 返回其层次遍历结果：
 *
 * [
 *   [3],
 *   [9,20],
 *   [15,7]
 * ]
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/cong-shang-dao-xia-da-yin-er-cha-shu-ii-lcof
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 * @author caihe
 * @date 2022/8/24 14:01
 */
public class _32II从上到下打印二叉树II {

    List<List<Integer>> resList = new ArrayList<>();

    /**
     * 使用前序遍历遍历所有的节点
     *  需要通过deep变量获取遍历的深度 然后根据深度和List设置节点值
     */
    public List<List<Integer>> levelOrder(TreeNode root) {
        fun(root, 0);
        return resList;
    }
    
    public void fun(TreeNode root, int deep) {
        if (root == null) {
            return;
        }
        if (resList.size() <= deep) {
            resList.add(new ArrayList<>());
        }
        List<Integer> list = resList.get(deep);
        if (list == null) {
            list = new ArrayList<>();
            resList.set(deep, list);
        }
        list.add(root.val);
        fun(root.left, deep + 1);
        fun(root.right, deep + 1);
    }


    /**
     * bfs版本
     */
    public List<List<Integer>> levelOrder_bfs(TreeNode root) {
        if (root == null) {
            return  resList;
        }
        Queue<TreeNode> queue = new LinkedList<>();
        queue.add(root);
        int deep = 0;
        while(!queue.isEmpty()) {
            List<Integer> list = new ArrayList<>();
            // 消费掉每层的节点
            Queue<TreeNode> newQueue = new LinkedList<>();
            while(!queue.isEmpty()) {
                newQueue.add(queue.poll());
            }
            for (TreeNode treeNode : newQueue) {
                list.add(treeNode.val);
                if (treeNode.left != null) {
                    queue.add(treeNode.left);
                } 
                if (treeNode.right != null) {
                    queue.add(treeNode.right);
                }
            }
            resList.add(list);
        }
        return resList;
    }
    
    
}
